# -*- coding: utf-8 -*-
import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':

    # Load a model
    model = YOLO(model=r'./runs/detect/train/weights/best.pt')  # segment   / classify  /  detect
    model.val(data='data.yaml',
                imgsz=640,
                batch=16,
                split='test',
                workers=10,
                device='0',
                  )

